#ifndef DETECTMASK_H
#define DETECTMASK_H
#include <iostream>
#include <string>
#include <opencv2/opencv.hpp>
#include "paddle_api.h"

class DetectMask
{
public:
    DetectMask();


    bool init(std::string det_model_dir, std::string class_model_dir);
    bool detect(cv::Mat img);

    std::shared_ptr<paddle::lite_api::PaddlePredictor> facepredictor;
    std::shared_ptr<paddle::lite_api::PaddlePredictor> maskpredictor;

};

#endif // DETECTMASK_H
